package leetcode.greedy;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 621. 任务调度器
 * 给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间，CPU 可以完成一个任务，或者处于待命状态。
 * <p>
 * 然而，两个 相同种类 的任务之间必须有长度为整数 n 的冷却时间，因此至少有连续 n 个单位时间内 CPU 在执行不同的任务，或者在待命状态。
 * <p>
 * 你需要计算完成所有任务所需要的 最短时间 。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：tasks = ["A","A","A","B","B","B"], n = 2
 * 输出：8
 * 解释：A -> B -> (待命) -> A -> B -> (待命) -> A -> B
 * 在本示例中，两个相同类型任务之间必须间隔长度为 n = 2 的冷却时间，而执行一个任务只需要一个单位时间，所以中间出现了（待命）状态。
 * 示例 2：
 * <p>
 * 输入：tasks = ["A","A","A","B","B","B"], n = 0
 * 输出：6
 * 解释：在这种情况下，任何大小为 6 的排列都可以满足要求，因为 n = 0
 * ["A","A","A","B","B","B"]
 * ["A","B","A","B","A","B"]
 * ["B","B","B","A","A","A"]
 * ...
 * 诸如此类
 * 示例 3：
 * <p>
 * 输入：tasks = ["A","A","A","A","A","A","B","C","D","E","F","G"], n = 2
 * 输出：16
 * 解释：一种可能的解决方案是：
 * A -> B -> C -> A -> D -> E -> A -> F -> G -> A -> (待命) -> (待命) -> A -> (待命) -> (待命) -> A
 */
public class LeastInterval {

    public int leastInterval(char[] tasks, int n) {

        Map<Character, Integer> map = new HashMap<>();
        // 最多的一个字符有 max 个。
        int max = 0;
        for (char c : tasks) {
            int num = map.getOrDefault(c, 0) + 1;
            map.put(c, num);
            max = Math.max(max, num);
        }

        int maxCount = 0;
        Set<Map.Entry<Character, Integer>> set = map.entrySet();
        for (Map.Entry<Character, Integer> entry : set) {
            // val 出现的次数
            int val = entry.getValue();
            if (val == max) {
                ++maxCount;
            }
        }

        //  A A A A B B B B C C D E ,n = 2 为例：
        // [A * *] [A * *] [A * *] + maxCount
        // maxCount 为几那么在填不满的情况下最后一组[A * *]之后就有几个字母，其中肯定包含一个A。比如此例就是A B
        // 如果其他元素能填满A之间的空格 那么返回task.length；如果填不满返回 Math.max(((max - 1) * (n + 1) + maxCount)

        return Math.max(
                ((max - 1) * (n + 1) + maxCount),
                tasks.length
        );
    }
}
